Have to fix my problem, this should work properly