fixed stupid code :)