Fixed one more bug :)