Fixed an error with compilation, but it works now