small edit should work