Fix embarassing bug