Small change but this might be better