Fixed bug, works now