Fixed a cleanup / stubbing error.