Fixed the problem with it not working