Update to include spec helper