Fixed build-breaking bug