fix last minute code change