Fixing controller generation script