Updated to fix lint test