Small bug fix/MIT License