Fixed "'module' not callable" bug