Add allocation test