Fixed dem bugs