Correctify bug