fix last minute bug