Fix hard code