Bug fix with indentation