Not sure what needs to be fixed? This looks like the right approach.