Fixed the bug lmao